package com.abss.vote.util;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.internal.DefaultShellCallback;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

/**
 * 生成实体类
 */
public class MybatisUtil {


    //该方法用于自动生成dao model mapper文件
    public static void main(String[] args){

        try {
            GenerateAllFile();
        }catch (Exception e){
            e.printStackTrace();
        }

    }

    public static void GenerateAllFile() throws Exception{
        List<String> warnings = new ArrayList<String>();
        boolean overwrite = true;
        File configFile = new File("./src/main/resources/mybatis/generatorConfig.xml");
        if(!configFile.exists()){
            System.out.println("不存在文件");
            return;
        }

        ConfigurationParser cp = new ConfigurationParser(warnings);
        Configuration config = cp.parseConfiguration(configFile);
        DefaultShellCallback callback = new DefaultShellCallback(overwrite);


        MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
        myBatisGenerator.generate(null);

    }
}
